#!/bin/bash
# --------------------------------------------------------------------------------------------------
# Analysis_Python_Controller
#
# This backend bash script manages the dispatch of all analysis jobs that run in Python
# --------------------------------------------------------------------------------------------------

# INPUT ENVIRONMENT VARIABLES
echo "USER="${1}
echo "STEP="${2}

# OUTPUT ENVIRONMENT VARIABLES
echo "SLURM_JOB_ID="$SLURM_JOB_ID
echo "SLURM_JOB_NODELIST"=$SLURM_JOB_NODELIST
echo "SLURM_NNODES"=$SLURM_NNODES
echo "SLURM_SUBMIT_DIR="$SLURM_SUBMIT_DIR
echo "SLURM_ARRAY_TASK_ID="$SLURM_ARRAY_TASK_ID
echo "SLURM_ARRAY_JOB_ID"=$SLURM_ARRAY_JOB_ID
echo "SLURM_NTASKS"=$SLURM_NTASKS

# LOAD SHELL PROFILE
if [ -f ~/shell_profile.sh ];
then
    source ~/shell_profile.sh
else
    echo "Please set up ~/shell_profile.sh file. Exiting."
    exitcd
fi

# RUN CALCULATIONS
case ${2} in
  "Security_Regressions")
    python "${code_path}/analysis/security_regressions.py"
  ;;
  "NonUS_Estimates")
    python "${code_path}/analysis/nonus_estimates.py"
  ;;
esac

# FINISHED
echo "Finished Step "${2}
exit
